{'flow_embedding': tensor([[ 9.6398e-01, -2.3675e-01,  1.5377e+00,  3.5464e-01, -5.5787e-01,
          4.7373e-01,  1.1232e+00,  6.7407e-01,  1.7883e-01,  1.2046e+00,
          4.5138e-01, -9.3654e-01,  9.3287e-01, -3.4683e-01,  4.7318e-01,
         -3.8474e-01, -1.0964e+00, -3.7018e-01, -3.4102e-01, -9.4742e-01,
         -1.0389e-04, -7.5092e-01, -3.5038e-02,  2.2328e+00, -7.8314e-01,
          1.3110e+00,  6.3498e-01,  1.7557e+00, -2.1962e-01, -3.4139e-01,
         -1.7386e+00, -7.4859e-01, -2.5943e-01, -2.1374e+00, -5.8032e-01,
         -2.4653e+00, -1.1978e+00,  8.7352e-01,  7.7237e-01,  8.0546e-01,
          1.1365e+00, -8.9831e-01,  8.1807e-01, -2.9690e-01,  1.4736e+00,
          2.6340e-01,  1.9852e-01,  4.4634e-01,  4.8917e-01, -8.8731e-01,
          4.9187e-01, -1.6210e+00, -1.5089e+00, -2.3864e-01, -9.2280e-02,
         -7.7235e-01,  1.8202e-01,  3.6218e-01,  8.6071e-01,  1.7969e-02,
         -4.5983e-01,  6.8914e-01, -8.7130e-01, -3.5636e-01, -1.6557e-01,
         -3.9650e-01, -5.2790e-01, -2.8673e-01,  5.2943e-01,  1.7909e+00,
          3.0601e-01,  1.3771e+00,  1.3534e-01,  6.7058e-01, -3.5119e-02,
         -3.2232e-01,  2.4329e-01,  2.9949e-01,  9.9140e-01,  1.1219e-01,
         -7.8075e-01,  1.0671e+00, -1.0086e+00,  2.3732e+00, -4.2324e-01,
         -7.8819e-01, -4.1830e-01, -1.2814e+00, -1.7419e-01, -1.6930e+00,
          4.7747e-01,  7.2026e-01, -1.1067e+00, -9.3431e-01, -1.5040e-01,
         -2.1000e-01, -1.3335e+00, -1.2730e+00, -3.2893e-01,  3.0202e-01,
          3.7001e-01, -7.8176e-01,  3.7907e-01,  1.6932e+00, -4.2775e-01,
         -3.9432e-01, -8.4945e-02,  1.8923e+00,  1.3116e+00, -1.1401e+00,
          9.4482e-01,  1.8643e+00, -1.4221e+00, -3.4329e-02, -9.6312e-01,
         -8.4378e-01,  1.5074e-01,  3.5687e-01, -2.3118e-01,  4.4479e-01,
          1.2592e-01, -1.8482e+00, -6.6402e-01, -1.1042e+00, -9.3434e-01,
         -1.4556e+00,  1.0859e+00, -7.1755e-01,  4.3009e-01, -7.8765e-01,
         -9.6422e-01, -8.7007e-01,  2.5427e-01, -4.4904e-01,  8.8142e-01,
          6.5395e-01,  6.6288e-01,  6.6787e-01,  3.8196e-01,  3.0491e-01,
          1.0761e+00, -9.2298e-01, -1.0651e+00,  2.5811e-01,  2.0605e+00,
         -1.0111e-01,  4.8077e-01, -8.1497e-01,  6.6769e-01,  3.0165e-01,
         -2.0793e+00,  9.4994e-01, -3.3611e-01, -5.9973e-02,  1.2164e+00,
         -1.3834e+00, -3.8607e-01, -2.0280e-02, -8.8003e-01, -1.9313e-01,
          4.4198e-01,  2.5334e-01, -4.1108e-01, -1.3954e-01,  3.3478e-01,
          1.4501e-01,  8.1999e-03,  2.1478e-01,  4.2633e-01, -1.3904e+00,
          3.0012e-01,  8.9084e-01,  8.3920e-01, -9.6367e-01, -7.9073e-01,
         -3.8909e-01,  4.1772e-01, -1.4161e+00,  1.7143e-01, -3.4268e-01,
          4.4849e-01,  1.4519e+00, -7.0847e-01, -1.8272e+00,  1.0210e+00,
         -5.6029e-01, -4.9288e-01, -8.7554e-02, -3.0222e-01,  2.5673e-01,
          2.9467e-02, -7.6578e-03]], device='cuda:0'), 'llm_embedding': tensor([[ 9.6398e-01, -2.3675e-01,  1.5377e+00,  3.5464e-01, -5.5787e-01,
          4.7373e-01,  1.1232e+00,  6.7407e-01,  1.7883e-01,  1.2046e+00,
          4.5138e-01, -9.3654e-01,  9.3287e-01, -3.4683e-01,  4.7318e-01,
         -3.8474e-01, -1.0964e+00, -3.7018e-01, -3.4102e-01, -9.4742e-01,
         -1.0389e-04, -7.5092e-01, -3.5038e-02,  2.2328e+00, -7.8314e-01,
          1.3110e+00,  6.3498e-01,  1.7557e+00, -2.1962e-01, -3.4139e-01,
         -1.7386e+00, -7.4859e-01, -2.5943e-01, -2.1374e+00, -5.8032e-01,
         -2.4653e+00, -1.1978e+00,  8.7352e-01,  7.7237e-01,  8.0546e-01,
          1.1365e+00, -8.9831e-01,  8.1807e-01, -2.9690e-01,  1.4736e+00,
          2.6340e-01,  1.9852e-01,  4.4634e-01,  4.8917e-01, -8.8731e-01,
          4.9187e-01, -1.6210e+00, -1.5089e+00, -2.3864e-01, -9.2280e-02,
         -7.7235e-01,  1.8202e-01,  3.6218e-01,  8.6071e-01,  1.7969e-02,
         -4.5983e-01,  6.8914e-01, -8.7130e-01, -3.5636e-01, -1.6557e-01,
         -3.9650e-01, -5.2790e-01, -2.8673e-01,  5.2943e-01,  1.7909e+00,
          3.0601e-01,  1.3771e+00,  1.3534e-01,  6.7058e-01, -3.5119e-02,
         -3.2232e-01,  2.4329e-01,  2.9949e-01,  9.9140e-01,  1.1219e-01,
         -7.8075e-01,  1.0671e+00, -1.0086e+00,  2.3732e+00, -4.2324e-01,
         -7.8819e-01, -4.1830e-01, -1.2814e+00, -1.7419e-01, -1.6930e+00,
          4.7747e-01,  7.2026e-01, -1.1067e+00, -9.3431e-01, -1.5040e-01,
         -2.1000e-01, -1.3335e+00, -1.2730e+00, -3.2893e-01,  3.0202e-01,
          3.7001e-01, -7.8176e-01,  3.7907e-01,  1.6932e+00, -4.2775e-01,
         -3.9432e-01, -8.4945e-02,  1.8923e+00,  1.3116e+00, -1.1401e+00,
          9.4482e-01,  1.8643e+00, -1.4221e+00, -3.4329e-02, -9.6312e-01,
         -8.4378e-01,  1.5074e-01,  3.5687e-01, -2.3118e-01,  4.4479e-01,
          1.2592e-01, -1.8482e+00, -6.6402e-01, -1.1042e+00, -9.3434e-01,
         -1.4556e+00,  1.0859e+00, -7.1755e-01,  4.3009e-01, -7.8765e-01,
         -9.6422e-01, -8.7007e-01,  2.5427e-01, -4.4904e-01,  8.8142e-01,
          6.5395e-01,  6.6288e-01,  6.6787e-01,  3.8196e-01,  3.0491e-01,
          1.0761e+00, -9.2298e-01, -1.0651e+00,  2.5811e-01,  2.0605e+00,
         -1.0111e-01,  4.8077e-01, -8.1497e-01,  6.6769e-01,  3.0165e-01,
         -2.0793e+00,  9.4994e-01, -3.3611e-01, -5.9973e-02,  1.2164e+00,
         -1.3834e+00, -3.8607e-01, -2.0280e-02, -8.8003e-01, -1.9313e-01,
          4.4198e-01,  2.5334e-01, -4.1108e-01, -1.3954e-01,  3.3478e-01,
          1.4501e-01,  8.1999e-03,  2.1478e-01,  4.2633e-01, -1.3904e+00,
          3.0012e-01,  8.9084e-01,  8.3920e-01, -9.6367e-01, -7.9073e-01,
         -3.8909e-01,  4.1772e-01, -1.4161e+00,  1.7143e-01, -3.4268e-01,
          4.4849e-01,  1.4519e+00, -7.0847e-01, -1.8272e+00,  1.0210e+00,
         -5.6029e-01, -4.9288e-01, -8.7554e-02, -3.0222e-01,  2.5673e-01,
          2.9467e-02, -7.6578e-03]], device='cuda:0')}